#!/usr/bin/env python
"""
GT-LMS 测试数据生成运行脚本
用于运行数据生成器并生成测试数据
"""

import sys
import os
from pathlib import Path

# 添加项目根目录到Python路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from data.data_generator import DataGenerator


def main():
    """主函数"""
    print("GT-LMS 测试数据生成器")
    print("=" * 50)
    
    # 创建数据生成器
    generator = DataGenerator()
    
    # 生成测试数据
    data = generator.generate_all_data(
        num_admins=2,      # 管理员数量
        num_teachers=6,    # 教师数量
        num_students=40,   # 学生数量
        courses_per_teacher=3,  # 每位教师开设的课程数量
        min_courses_per_student=3,  # 每个学生最少选课数
        max_courses_per_student=5,  # 每个学生最多选课数
        assignments_per_course=4,  # 每门课程的作业数量
        submission_rate=0.85,  # 提交率
        late_rate=0.15      # 迟交率
    )
    
    # 保存数据到文件
    success = generator.save_data_to_files(data)
    
    if success:
        # 打印统计信息
        generator.print_statistics(data)
        print("\n测试数据生成并保存成功！")
        print(f"数据已保存到: {generator.data_dir}")
    else:
        print("\n测试数据保存失败！")


if __name__ == "__main__":
    main()